from typing import Union

from fastapi import FastAPI,File, UploadFile

from ocr import ocr



app = FastAPI()

ocrEngine = ocr()

@app.get("/")
def read_root():
    return {"Hello": "World"}


@app.get("/items/{item_id}")
def read_item(item_id: int, q: Union[str, None] = None):
    return {"item_id": item_id, "q": q}

@app.post("/ocrImage/")
async def ocr_image(file: UploadFile):
    print(file.content_type)
    if file.content_type.count('image') != 0:
        return ocrEngine.getRes(file.file.read())
    else:
        return 'no'

#python -m uvicorn main:app --reload